from poolnet import PoolNet
from deeplab_resnet import resnet18_locate
if __name__ == '__main__':
    model = PoolNet('resnet18', resnet18_locate(), None)
    model.eval()
    input = torch.randn(1, 3, 512, 512)
    output = model(input)
    print(output.size())
